python for loop advanced